package com.study.leetcode;

public class Solution_198 {
    /**
     *
     * @param nums
     * @return
     *
     * dp[i][0] 代表当前节点不选择
     * dp[i][0] 代表当前节点选择
     */
    public static int rob(int[] nums) {
        int[][] dp = new int[nums.length][2];
        dp[0][1] = nums[0];
        for (int i = 1; i < nums.length; i++) {
            dp[i][0] = Math.max(dp[i - 1][1], dp[i - 1][0]);
            dp[i][1] = Math.max(dp[i - 1][0] + nums[i], dp[i - 1][0]);
        }
        return Math.max(dp[nums.length - 1][0], dp[nums.length - 1][1]);
    }

    /**
     *
     * @param nums
     * @return
     * dp[i] 代表当前节点 则当前节点的值来自于： 如果当前节点不选择则上一个节点必然是选择的
     * ，如果选择当前节点则 上一个节点不能选择必须是选择上上一个节点
     */
    public static int rob1(int[] nums) {
        if (nums.length <= 1) {
            return nums[0];
        }
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[1],dp[0]);
        for (int i = 2; i < nums.length; i++) {
            dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
        }
        return dp[nums.length - 1];
    }

    /**
     *
     * @param nums
     * @return
     * 将上一个算法化简则得到如下
     */
    public static int rob2(int[] nums) {
        if (nums.length <= 1) {
            return nums[0];
        }
        int f0 = 0;
        int f1 = 0;
        int newF;
        for (int num : nums) {
            newF = Math.max(f0 + num, f1);
            f0 = f1;
            f1 = newF;

        }
        return f1;
    }

    public static void main(String[] args) {
        int[] nums = {1,2,3,1};
        System.out.println("job = " + rob(nums));
        System.out.println("job = " + rob1(nums));
        System.out.println("job = " + rob2(nums));

    }
}
